/* SPDX-License-Identifier: GPL-2.0-or-later */
/* Copyright(c) 2020 - 2023 Allwinner Technology Co.,Ltd. All rights reserved. */
/*
 * SUNXI MTD partitioning
 *
 * Copyright (c) 2016 WimHuang <huangwei@allwinnertech.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 *
 */


#ifndef FS_PART_EFI_H_INCLUDED
#define FS_PART_EFI_H_INCLUDED

#include <linux/types.h>
#include <linux/fs.h>
#include <linux/kernel.h>
#include <linux/major.h>
#include <linux/string.h>
#include <linux/efi.h>
#include <linux/compiler.h>
#include <linux/version.h>
#if (LINUX_VERSION_CODE < KERNEL_VERSION(6, 1, 12))
#include <linux/genhd.h>
#endif

#define MSDOS_MBR_SIGNATURE 0xaa55
#define EFI_PMBR_OSTYPE_EFI 0xEF
#define EFI_PMBR_OSTYPE_EFI_GPT 0xEE

#define GPT_MBR_PROTECTIVE 1
#define GPT_MBR_HYBRID 2

#define GPT_HEADER_SIGNATURE 0x5452415020494645ULL
#define GPT_HEADER_REVISION_V1 0x00010000
#define GPT_PRIMARY_PARTITION_TABLE_LBA 1

#define PARTITION_SYSTEM_GUID                                                  \
	EFI_GUID(0xC12A7328, 0xF81F, 0x11d2, 0xBA, 0x4B, 0x00, 0xA0, 0xC9,     \
		 0x3E, 0xC9, 0x3B)
#define LEGACY_MBR_PARTITION_GUID                                              \
	EFI_GUID(0x024DEE41, 0x33E7, 0x11d3, 0x9D, 0x69, 0x00, 0x08, 0xC7,     \
		 0x81, 0xF3, 0x9F)
#define PARTITION_MSFT_RESERVED_GUID                                           \
	EFI_GUID(0xE3C9E316, 0x0B5C, 0x4DB8, 0x81, 0x7D, 0xF9, 0x2D, 0xF0,     \
		 0x02, 0x15, 0xAE)
#define PARTITION_BASIC_DATA_GUID                                              \
	EFI_GUID(0xEBD0A0A2, 0xB9E5, 0x4433, 0x87, 0xC0, 0x68, 0xB6, 0xB7,     \
		 0x26, 0x99, 0xC7)
#define PARTITION_LINUX_RAID_GUID                                              \
	EFI_GUID(0xa19d880f, 0x05fc, 0x4d3b, 0xa0, 0x06, 0x74, 0x3f, 0x0f,     \
		 0x84, 0x91, 0x1e)
#define PARTITION_LINUX_SWAP_GUID                                              \
	EFI_GUID(0x0657fd6d, 0xa4ab, 0x43c4, 0x84, 0xe5, 0x09, 0x33, 0xc8,     \
		 0x4b, 0x4f, 0x4f)
#define PARTITION_LINUX_LVM_GUID                                               \
	EFI_GUID(0xe6d6d379, 0xf507, 0x44c2, 0xa2, 0x3c, 0x23, 0x8f, 0x2a,     \
		 0x3d, 0xf9, 0x28)

struct _gpt_header {
	__le64 signature;
	__le32 revision;
	__le32 header_size;
	__le32 header_crc32;
	__le32 reserved1;
	__le64 my_lba;
	__le64 alternate_lba;
	__le64 first_usable_lba;
	__le64 last_usable_lba;
	efi_guid_t disk_guid;
	__le64 partition_entry_lba;
	__le32 num_partition_entries;
	__le32 sizeof_partition_entry;
	__le32 partition_entry_array_crc32;

	/* The rest of the logical block is reserved by UEFI and must be zero.
	 * EFI standard handles this by:
	 *
	 * uint8_t		reserved2[ BlockSize - 92 ];
	 */
} __packed;

struct _gpt_entry_attributes {
	u64 required_to_function : 1;
	u64 reserved : 47;
	u64 type_guid_specific : 16;
} __packed;

struct _gpt_entry {
	efi_guid_t partition_type_guid;
	efi_guid_t unique_partition_guid;
	__le64 starting_lba;
	__le64 ending_lba;
	struct _gpt_entry_attributes attributes;
	efi_char16_t partition_name[72 / sizeof(efi_char16_t)];
} __packed;

struct _gpt_mbr_record {
	u8 boot_indicator; /* unused by EFI, set to 0x80 for bootable */
	u8 start_head;     /* unused by EFI, pt start in CHS */
	u8 start_sector;   /* unused by EFI, pt start in CHS */
	u8 start_track;
	u8 os_type;	  /* EFI and legacy non-EFI OS types */
	u8 end_head;	 /* unused by EFI, pt end in CHS */
	u8 end_sector;       /* unused by EFI, pt end in CHS */
	u8 end_track;	/* unused by EFI, pt end in CHS */
	__le32 starting_lba; /* used by EFI - start addr of the on disk pt */
	__le32 size_in_lba;  /* used by EFI - size of pt in LBA */
} __packed;

struct _legacy_mbr {
	u8 boot_code[440];
	__le32 unique_mbr_signature;
	__le16 unknown;
	struct _gpt_mbr_record partition_record[4];
	__le16 signature;
} __packed;

#endif
